package com.ruoyi.framework.config;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashSet;
import java.util.List;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.io.Resource;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
import org.springframework.core.io.support.ResourcePatternResolver;
import org.springframework.core.type.classreading.CachingMetadataReaderFactory;
import org.springframework.core.type.classreading.MetadataReader;
import org.springframework.core.type.classreading.MetadataReaderFactory;
import org.springframework.util.ClassUtils;
/**
 * Mybatis支持*匹配扫描包
 * 
 * @author ruoyi
 */
@Configuration
public class MyBatisConfig
{
	//    @Autowired
	//    private Environment env;

	static final String DEFAULT_RESOURCE_PATTERN = "**/*.class";

	public static String setTypeAliasesPackage(String typeAliasesPackage)
	{
		ResourcePatternResolver resolver = (ResourcePatternResolver) new PathMatchingResourcePatternResolver();
		MetadataReaderFactory metadataReaderFactory = new CachingMetadataReaderFactory(resolver);
		List<String> allResult = new ArrayList<String>();
		try
		{
			for (String aliasesPackage : typeAliasesPackage.split(","))
			{
				List<String> result = new ArrayList<String>();
				aliasesPackage = ResourcePatternResolver.CLASSPATH_ALL_URL_PREFIX
						+ ClassUtils.convertClassNameToResourcePath(aliasesPackage.trim()) + "/" + DEFAULT_RESOURCE_PATTERN;
				Resource[] resources = resolver.getResources(aliasesPackage);
				if (resources != null && resources.length > 0)
				{
					MetadataReader metadataReader = null;
					for (Resource resource : resources)
					{
						if (resource.isReadable())
						{
							metadataReader = metadataReaderFactory.getMetadataReader(resource);
							try
							{
								result.add(Class.forName(metadataReader.getClassMetadata().getClassName()).getPackage().getName());
							}
							catch (ClassNotFoundException e)
							{
								e.printStackTrace();
							}
						}
					}
				}
				if (result.size() > 0)
				{
					HashSet<String> hashResult = new HashSet<String>(result);
					allResult.addAll(hashResult);
				}
			}
			if (allResult.size() > 0)
			{
				typeAliasesPackage = String.join(",", (String[]) allResult.toArray(new String[0]));
			}
			else
			{
				throw new RuntimeException("mybatis typeAliasesPackage 路径扫描错误,参数typeAliasesPackage:" + typeAliasesPackage + "未找到任何包");
			}
		}
		catch (IOException e)
		{
			e.printStackTrace();
		}
		return typeAliasesPackage;
	}

	public Resource[] resolveMapperLocations(String[] mapperLocations)
	{
		ResourcePatternResolver resourceResolver = new PathMatchingResourcePatternResolver();
		List<Resource> resources = new ArrayList<Resource>();
		if (mapperLocations != null)
		{
			for (String mapperLocation : mapperLocations)
			{
				try
				{
					Resource[] mappers = resourceResolver.getResources(mapperLocation);
					resources.addAll(Arrays.asList(mappers));
				}
				catch (IOException e)
				{
					// ignore
				}
			}
		}
		return resources.toArray(new Resource[resources.size()]);
	}


	//    @Bean
	//    public SqlSessionFactory sqlSessionFactory(DataSource dataSource) throws Exception
	//    {
	//        String typeAliasesPackage = env.getProperty("mybatis.typeAliasesPackage");
	//        String mapperLocations = env.getProperty("mybatis.mapperLocations");
	//        String configLocation = env.getProperty("mybatis.configLocation");
	//        typeAliasesPackage = setTypeAliasesPackage(typeAliasesPackage);
	//        VFS.addImplClass(SpringBootVFS.class);
	//
	//        final SqlSessionFactoryBean sessionFactory = new SqlSessionFactoryBean();
	//        sessionFactory.setDataSource(dataSource);
	//        sessionFactory.setTypeAliasesPackage(typeAliasesPackage);
	//        sessionFactory.setMapperLocations(resolveMapperLocations(StringUtils.split(mapperLocations, ",")));
	//        sessionFactory.setConfigLocation(new DefaultResourceLoader().getResource(configLocation));
	//        return sessionFactory.getObject();
	//    }
}